package com.dingreading.cloud.common.util;

/**
 * 获取文件的查看类型
 */
public class FileViewUtil {

    /**
     * 根据文件的扩展名获取类型名
     */
    public static int getViewType(String fileName) {
        if (fileName.endsWith(".mp4") || fileName.endsWith(".MP4")) {
            return 1;
        } else if (fileName.endsWith(".rar") || fileName.endsWith(".zip")) {
            return 2;
        } else if (fileName.endsWith(".jpg") || fileName.endsWith(".png") || fileName.endsWith(".jpeg")) {
            return 3;
        } else if (fileName.endsWith(".pdf") || fileName.endsWith(".PDF")) {
            return 4;
        }
        return 0;
    }

}
